from iop import Message
from dataclasses import dataclass
from langchain.docstore.document import Document

@dataclass
class ChunkMsg(Message):
    docs: list[Document]=None
    
@dataclass
class GeneralMsg(Message):
    msg: str=None